package com.will.nio.selector;

import lombok.extern.slf4j.Slf4j;

import java.io.IOException;
import java.net.InetSocketAddress;
import java.nio.channels.*;
import java.util.Iterator;

/**
 * @author Will
 * @description:
 *              Selector    -   处理 accept
 * @date 2023-04-13 19:34
 * @version: 1.0
 */
@Slf4j
public class AcceptSelectorServer {
    public static void main(String[] args) throws IOException {
        // 1. 创建 selector，管理多个 channel
        Selector selector = Selector.open();
        ServerSocketChannel ssc = ServerSocketChannel.open();
        ssc.configureBlocking(false);
        // 2. 建立 selector 和 channel 的联系（注册），channel 必须工作在非阻塞模式下
        // SelectionKey 就是将来事件发生后，通过它可以知道什么事件和哪个 channel 发生的事件
        SelectionKey sscKey = ssc.register(selector, 0, null);
        log.debug("register key:{}", sscKey);
        // 只关注 accept 事件
        sscKey.interestOps(SelectionKey.OP_ACCEPT);
        ssc.bind(new InetSocketAddress(8080));
        while (true) {
            // 3. select 方法，没有事件发生，线程阻塞，有事件，线程才会恢复运行
            selector.select();
            // 4. 处理事件，selectedKeys 内部包含了所有发生的事件
            Iterator<SelectionKey> iterator = selector.selectedKeys().iterator();
            while (iterator.hasNext()) {
                SelectionKey key = iterator.next();
                log.debug("key:{}", key);
                ServerSocketChannel channel = (ServerSocketChannel)key.channel();
                SocketChannel sc = channel.accept();
                log.debug("{}", sc);
            }
        }
    }
}
